python 机器学习笔记
构建模型
决策树
from sklearn.tree import DecisionTreeRegressor
model = DecisionTreeRegressor(random_state=1, max_depth=100000)
随机森林
from sklearn.ensemble import RandomForestRegressor
训练模型
model.fit(a_x, a_y)
预测数据
predictions = model.predict(b_x)
数据拆分和计算模型准确度
导入平均绝对误差(Mean Absolute Error,MAE)库
from sklearn.metrics import mean_absolute_error
导入数据拆分库
from sklearn.model_selection import train_test_split
拆分数据
a_x, b_x, a_y, b_y = train_test_split(x, Y, random_state=1)
拆分后的数据a_x, b_x, a_y, b_y 是原数据x,Y的子集,一部分用来训练,一部分用来验证
计算平均绝对误差
detla = mean_absolute_error(b_y, predictions)
predictions是预测出来的数据,b_y是实际的数据,函数返回差值
本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 AndrewLee!